package com.zhang.hadoop.netty.netty.groupchat;

import io.netty.channel.Channel;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.SimpleChannelInboundHandler;
import io.netty.channel.group.ChannelGroup;
import io.netty.channel.group.DefaultChannelGroup;
import io.netty.util.concurrent.GlobalEventExecutor;

import java.util.HashMap;
import java.util.Map;

/**
 * @author zhang yufei
 * @date 2022/2/10 6:54 下午
 **/
public class GroupChatServerHandler extends SimpleChannelInboundHandler<String> {

    //使用一个hashMao管理
    public static Map<User, Channel> channels = new HashMap<>();

    //定义一个channel组,管理所有的channel
    //GlobalEventExecutor.INSTANCE是全局的事件执行器,是一个单例
    private static ChannelGroup channelGroup = new DefaultChannelGroup(GlobalEventExecutor.INSTANCE);

    //handlerAdded表示建立连接,一旦连接,第一个背执行
    //将当前channel加入到channelGroup
    @Override
    public void handlerAdded(ChannelHandlerContext ctx) {
        Channel channel = ctx.channel();
        //将该客户加入俩天的信息推送给其他在线的客户
        channelGroup.writeAndFlush("客户端:" + channel.remoteAddress() + "加入聊天");
        channelGroup.add(channel);
        //发送给指定用户
        channels.put(new User(1,"123"),channel);
    }

    //断开链接,将xx客户离开信息推送给当前在线的客户
    @Override
    public void handlerRemoved(ChannelHandlerContext ctx) throws Exception {
        Channel channel = ctx.channel();
        channelGroup.writeAndFlush("客户端" + channel.remoteAddress() + "离开了");
        System.out.println("channelGroup size:" + channelGroup.size());
    }

    //表示channel处于活动状态,提示xx上线
    @Override
    public void channelActive(ChannelHandlerContext ctx) throws Exception {
        System.out.println(ctx.channel().remoteAddress() + "上线了");
    }

    //表示channel处于不活动状态,提示xx离线了
    @Override
    public void channelInactive(ChannelHandlerContext ctx) throws Exception {
        System.out.println(ctx.channel().remoteAddress() + "离线了");
    }

    //读取数据
    @Override
    protected void channelRead0(ChannelHandlerContext channelHandlerContext, String s) throws Exception {
        //获取到当前channel
        Channel channel = channelHandlerContext.channel();
        //这时便利channelGroup,根据不同的情况,回送不同的消息
        channelGroup.forEach(ch -> {
            if (channel != ch) { //不是当前的channel,转发消息
                ch.writeAndFlush("客户" + channel.remoteAddress() + "发送了消息" + s + "\n");
            } else {//回显自己发送消息给自己
                channel.writeAndFlush("自己发送了消息" + s + "\n");
            }
        });
    }

    @Override
    public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
        //关闭通道
        ctx.close();
    }
}
